import re

# 定义输入文件和输出文件的路径
input_file_path = r'E:\postgraduate\大论文\seg_train\长数据集 500轮\长_训练过程.txt'
output_file_path = r'E:\postgraduate\大论文\seg_train\长数据集 500轮\total_loss.txt'

# 读取输入文件内容
with open(input_file_path, 'r', encoding='utf-8') as file:  # 打开输入文件并读取内容
    content = file.read()  # 将文件内容读入变量content中

# 使用正则表达式提取所有的f_score值
f_scores = re.findall(r'total_loss=([\d.]+)', content)  # 查找所有匹配'f_score='后面的数字（包括小数）的字符串

# 将提取到的f_score值写入输出文件，每个值一行
with open(output_file_path, 'w', encoding='utf-8') as file:  # 打开输出文件并准备写入内容
    for score in f_scores:  # 遍历所有提取到的f_score值
        file.write(score + '\n')  # 将每个f_score值写入文件，每个值占一行
